'''
由于macOS不支持CUDA，利用 Apple Silicon 的 Metal Performance Shaders（MPS）：
对于搭载 Apple Silicon（如 M1、M2、M3 芯片）的 Mac，
PyTorch 自 1.13 起提供了实验性的 MPS 后端支持，可以用于加速 GPU 计算任务。
执行下列代码，如果 MPS 后端可用，将输出tensor((1.), device='mps:0')
'''
import torch
if torch.backends.mps.is_available():
    mps_device = torch.device("mps")
    x = torch.ones(1, device=mps_device)
    print(x)
else:
    print("MPS device not found.")